define return type of function in python